#include"log.hpp"
#include <iostream>
#include <string>
#include <strings.h>
#include <cstring>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <functional>
#include <cstdlib>
#include<pthread.h>
using namespace std;
struct Thread
{
   struct sockaddr_in server;
   int sockfd;
};
void* send_message(void *args)
{
    string message;
    struct Thread*td=static_cast<struct Thread*>(args);
    socklen_t len = sizeof(td->server);
    while(true)
    {
        cout << "Please Enter@ "<<"请输入"<<endl;
        getline(cin,message);
        sendto(td->sockfd,message.c_str(),message.size(),0,(const sockaddr*)&td->server,len);
    }
}
void* recv_message(void*args)
{
    struct Thread*td=static_cast<struct Thread*>(args);
    char buffer[1024];
    while(true)
    {
        memset(buffer, 0, sizeof(buffer));
        struct sockaddr_in temp;
        socklen_t len = sizeof(temp);
        ssize_t s=recvfrom(td->sockfd,buffer,sizeof(buffer),0,(struct sockaddr*)&temp,&len);
        if(s>0)
        {
            buffer[s]=0;
            cout<<buffer<<endl;
        }
    }
}
void Usage(std::string proc)
{
    cout << "\n\rUsage: " << proc << " serverip serverport\n"<< endl;
}
int main(int argc,char* argv[])
{
    if(argc!=3) 
    {
        Usage(argv[0]);
        exit(0);
    }
    string serverip=argv[1];
    uint16_t serverport=stoi(argv[2]);
    struct Thread td;
    bzero(&td.server, sizeof(td.server));
    td.server.sin_family = AF_INET;
    td.server.sin_port = htons(serverport); //?
    td.server.sin_addr.s_addr = inet_addr(serverip.c_str());
    socklen_t len = sizeof(td.server);
    td.sockfd=socket(AF_INET, SOCK_DGRAM, 0);
    if(td.sockfd<0)
    {
         cout << "socker error" << endl;
        return 1;
    }
    // client 要bind吗？要！只不过不需要用户显示的bind！一般有OS自由随机选择！
    // 一个端口号只能被一个进程bind，对server是如此，对于client，也是如此！
    // 其实client的port是多少，其实不重要，只要能保证主机上的唯一性就可以！
    // 系统什么时候给我bind呢？首次发送数据的时候
    

    //创建线程收发数据线程。改进互不影响
    pthread_t sender,recvr;
    pthread_create(&sender,nullptr,send_message,&td);
    pthread_create(&recvr,nullptr,recv_message,&td);


    pthread_join(sender,nullptr);
    pthread_join(recvr,nullptr);
    // string message;
    // char buffer[1024];
    // while(true)
    // {
    //     cout<<"请输入"<<endl;
    //     getline(cin,message);
    //     sendto(sockfd,message.c_str(),message.size(),0,(struct sockaddr*)&server,len);//发送到服务器
    //     struct sockaddr_in temp;
    //     socklen_t len = sizeof(temp);
    //     ssize_t s=recvfrom(sockfd,buffer,sizeof(buffer),0,(struct sockaddr*)&temp,&len);
    //     if(s > 0)
    //     {
    //         buffer[s] = 0;
    //         cout << buffer << endl;
    //     }
    // }
     close(td.sockfd);
     return 0;

}


